home *** CD-ROM | disk | FTP | other *** search
/ Celestin Apprentice 5 / Apprentice-Release5.iso / Source Code / C / Applications / Python 1.3.3 / Python 133 68K / Lib / mailbox.py < prev    next >
Text File  |  1996-05-20  |  4KB  |  169 lines
  1. #
  2. # A class to hand a unix-style or mmdf-style mailboxes
  3. #
  4. # Jack Jansen, CWI, March 1994.
  5. #
  6. import rfc822
  7. import os
  8. import regex
  9.  
  10. class _Mailbox:
  11.     def __init__(self, fp):
  12.         self.fp = fp
  13.         self.seekp = 0
  14.  
  15.     def seek(self, pos, whence=0):
  16.         if whence==1:        # Relative to current position
  17.             self.pos = self.pos + pos
  18.         if whence==2:        # Relative to file's end
  19.             self.pos = self.stop + pos
  20.         else:            # Default - absolute position
  21.             self.pos = self.start + pos
  22.  
  23.     def next(self):
  24.         while 1:
  25.             self.fp.seek(self.seekp)
  26.             try:
  27.                 self._search_start()
  28.             except EOFError:
  29.                 self.seekp = self.fp.tell()
  30.                 return None
  31.             start = self.fp.tell()
  32.             self._search_end()
  33.             self.seekp = stop = self.fp.tell()
  34.             if start <> stop:
  35.                 break
  36.         return rfc822.Message(_Subfile(self.fp, start, stop))
  37.  
  38. class _Subfile:
  39.     def __init__(self, fp, start, stop):
  40.         self.fp = fp
  41.         self.start = start
  42.         self.stop = stop
  43.         self.pos = self.start
  44.  
  45.     def read(self, length = None):
  46.         if self.pos >= self.stop:
  47.             return ''
  48.         if length is None:
  49.             length = self.stop - self.pos
  50.         self.fp.seek(self.pos)
  51.         self.pos = self.pos + length
  52.         return self.fp.read(length)
  53.  
  54.     def readline(self, length = None):
  55.         if self.pos >= self.stop:
  56.             return ''
  57.         if length is None:
  58.             length = self.stop - self.pos
  59.         self.fp.seek(self.pos)
  60.         data = self.fp.readline(length)
  61.         if len(data) < length:
  62.             length = len(data)
  63.         self.pos = self.pos + length
  64.         return data
  65.  
  66.     def tell(self):
  67.         return self.pos - self.start
  68.  
  69.     def seek(self, pos, whence=0):
  70.             if whence == 0:
  71.             self.pos = self.start + pos
  72.         elif whence == 1:
  73.             self.pos = self.pos + pos
  74.         elif whence == 2:
  75.             self.pos = self.stop + pos
  76.  
  77.     def close(self):
  78.         pass
  79.  
  80. class UnixMailbox(_Mailbox):
  81.     def _search_start(self):
  82.         while 1:
  83.             line = self.fp.readline()
  84.             if not line:
  85.                 raise EOFError
  86.             if line[:5] == 'From ':
  87.                 return
  88.  
  89.     def _search_end(self):
  90.         while 1:
  91.             pos = self.fp.tell()
  92.             line = self.fp.readline()
  93.             if not line:
  94.                 return
  95.             if line[:5] == 'From ':
  96.                 self.fp.seek(pos)
  97.                 return
  98.  
  99. class MmdfMailbox(_Mailbox):
  100.     def _search_start(self):
  101.         while 1:
  102.             line = self.fp.readline()
  103.             if not line:
  104.                 raise EOFError
  105.             if line[:5] == '\001\001\001\001\n':
  106.                 return
  107.  
  108.     def _search_end(self):
  109.         while 1:
  110.             pos = self.fp.tell()
  111.             line = self.fp.readline()
  112.             if not line:
  113.                 return
  114.             if line == '\001\001\001\001\n':
  115.                 self.fp.seek(pos)
  116.                 return
  117.  
  118. class MHMailbox:
  119.     def __init__(self, dirname):
  120.     pat = regex.compile('^[0-9][0-9]*$')
  121.     self.dirname = dirname
  122.     files = os.listdir(self.dirname)
  123.     self.boxes = []
  124.     for f in files:
  125.         if pat.match(f) == len(f):
  126.         self.boxes.append(f)
  127.  
  128.     def next(self):
  129.     if not self.boxes:
  130.         return None
  131.     fn = self.boxes[0]
  132.     del self.boxes[0]
  133.     fp = open(os.path.join(self.dirname, fn))
  134.     return rfc822.Message(fp)
  135.         
  136.     
  137. if __name__ == '__main__':
  138.     import posix
  139.     import time
  140.     import sys
  141.     import string
  142. ##    mbox = '/usr/mail/'+posix.environ['USER']
  143. ##    fp = open(mbox, 'r')
  144. ##    mb = UnixMailbox(fp)
  145.  
  146.     mbox = posix.environ['HOME']+'/Mail/inbox'
  147.     mb = MHMailbox(mbox)
  148.     
  149.     msgs = []
  150.     while 1:
  151.         msg = mb.next()
  152.         if not msg:
  153.             break
  154.         msgs.append(msg)
  155.     if len(sys.argv) > 1:
  156.         num = string.atoi(sys.argv[1])
  157.         print 'Message %d body:'%num
  158.         msg = msgs[num-1]
  159.         msg.rewindbody()
  160.         sys.stdout.write(msg.fp.read())
  161.         sys.exit(0)
  162.     print 'Mailbox',mbox,'has',len(msgs),'messages:'
  163.     for msg in msgs:
  164.         f = msg.getheader('from')
  165.         s = msg.getheader('subject')
  166.         d = (msg.getheader('date'))
  167.         print '%20.20s   %18.18s   %-30.30s'%(f, d[5:], s)
  168.         
  169.